from page import Page
from selenium import webdriver
import unittest, time


class TestCase(unittest.TestCase):
    def setUp(self) -> None:
        print('开始执行')
        self.driver = webdriver.Edge()

    def tearDown(self) -> None:
        print('执行结束')
        time.sleep(5)
        self.driver.quit()

    def test_baidu_01(self):
        po = Page(self.driver)
        po.open_url('https://www.baidu.com')
        po.sou_suo('北京时间')
        po.sou_btn()


if __name__ == '__main__':
    unittest.main()
